#include "oled.h"
#include "oledfont.h"

// // 定义时钟倍率（72MHz时钟为基准）
// #define CLOCK_RATIO (168.0 / 72.0) //！！这里需要改为当前时钟

// // 全局延时因子（基于时钟倍率）
// static uint8_t i2c_delay_factor = (uint8_t)(6 * CLOCK_RATIO); // 基础延时因子

/**********************************************
//IIC Delay  // 优化后的延时函数 !!若不加延时则I2C失败
**********************************************/
void I2C_Delay(void)
{
	// for (uint8_t i = 0; i < i2c_delay_factor; i++)
	// {
	// 	__NOP();
	// }
	__NOP();
	__NOP();
	__NOP();
	__NOP();
	__NOP();
	__NOP();
	__NOP();
	__NOP();
	__NOP();
	__NOP();
	__NOP();
	__NOP(); // 168MHz下最小12个延时
}

/**********************************************
//IIC Start
**********************************************/
void IIC_Start()
{
	OLED_SCLK_Set();
	OLED_SDIN_Set();
	I2C_Delay(); // 替换原有的Delay_us(5)

	OLED_SDIN_Clr();
	I2C_Delay(); // 替换原有的Delay_us(5)

	OLED_SCLK_Clr();
	I2C_Delay(); // 添加额外延时
}

/**********************************************
//IIC Stop
**********************************************/
void IIC_Stop()
{
	OLED_SCLK_Set();
	OLED_SDIN_Clr();
	I2C_Delay(); // 替换原有的Delay_us(5)

	OLED_SDIN_Set();
	I2C_Delay(); // 添加额外延时
}

/**********************************************
//IIC Wait Ack
**********************************************/
uint8_t IIC_Wait_Ack()
{
	OLED_SDIN_Set();
	OLED_SCLK_Set();
	I2C_Delay(); // 替换原有的Delay_us(5)

	if (OLED_READ_SDA() == GPIO_PIN_SET)
	{
		IIC_Stop();
		printf("I2C失败\r\n");
		return 1;
	}

	OLED_SCLK_Clr();
	I2C_Delay(); // 添加额外延时

	return 0;
}

/**********************************************
// IIC Write byte
**********************************************/
void Write_IIC_Byte(unsigned char IIC_Byte)
{
	unsigned char i;
	unsigned char m, da;
	da = IIC_Byte;
	OLED_SCLK_Clr();

	for (i = 0; i < 8; i++)
	{
		m = da;
		m = m & 0x80;
		if (m == 0x80)
		{
			OLED_SDIN_Set();
		}
		else
		{
			OLED_SDIN_Clr();
		}

		I2C_Delay(); // 替换原有的Delay_us(5)
		da = da << 1;

		OLED_SCLK_Set();
		I2C_Delay(); // 替换原有的Delay_us(5)

		OLED_SCLK_Clr();
		I2C_Delay(); // 添加额外延时
	}
}

/**********************************************
// IIC Write Command
**********************************************/
void Write_IIC_Command(unsigned char IIC_Command)
{
	IIC_Start();
	Write_IIC_Byte(0x78); // Slave address,SA0=0
	if (IIC_Wait_Ack())
	{ // 发送地址，检查ACK
		IIC_Stop();
		return; // 失败退出
	}
	// IIC_Wait_Ack();
	Write_IIC_Byte(0x00); // write command
	IIC_Wait_Ack();
	Write_IIC_Byte(IIC_Command);
	IIC_Wait_Ack();
	IIC_Stop();
}

/**********************************************
// IIC Write Data
**********************************************/
void Write_IIC_Data(unsigned char IIC_Data)
{
	IIC_Start();
	Write_IIC_Byte(0x78); // D/C#=0; R/W#=0
	IIC_Wait_Ack();
	Write_IIC_Byte(0x40); // write data
	IIC_Wait_Ack();
	Write_IIC_Byte(IIC_Data);
	IIC_Wait_Ack();
	IIC_Stop();
}

/**********************************************
// OLED WR Byte
**********************************************/
void OLED_WR_Byte(unsigned dat, unsigned cmd)
{
	if (cmd)
	{

		Write_IIC_Data(dat);
	}
	else
	{
		Write_IIC_Command(dat);
	}
}

/********************************************
// fill_Picture
********************************************/
void fill_picture(unsigned char fill_Data)
{
	unsigned char m, n;
	for (m = 0; m < 8; m++)
	{
		OLED_WR_Byte(0xb0 + m, 0); // page0-page1
		OLED_WR_Byte(0x00, 0);	   // low column start address
		OLED_WR_Byte(0x10, 0);	   // high column start address
		for (n = 0; n < 128; n++)
		{
			OLED_WR_Byte(fill_Data, 1);
		}
	}
}

// 坐标设置
void OLED_Set_Pos(unsigned char x, unsigned char y)
{
	OLED_WR_Byte(0xb0 + y, OLED_CMD);
	OLED_WR_Byte(((x & 0xf0) >> 4) | 0x10, OLED_CMD);
	OLED_WR_Byte((x & 0x0f), OLED_CMD);
}
// 开启OLED显示
void OLED_Display_On(void)
{
	OLED_WR_Byte(0X8D, OLED_CMD); // SET DCDC命令
	OLED_WR_Byte(0X14, OLED_CMD); // DCDC ON
	OLED_WR_Byte(0XAF, OLED_CMD); // DISPLAY ON
}
// 关闭OLED显示
void OLED_Display_Off(void)
{
	OLED_WR_Byte(0X8D, OLED_CMD); // SET DCDC命令
	OLED_WR_Byte(0X10, OLED_CMD); // DCDC OFF
	OLED_WR_Byte(0XAE, OLED_CMD); // DISPLAY OFF
}
// 清屏函数,清完屏,整个屏幕是黑色的,相当于未点亮
void OLED_Clear(void)
{
	u8 i, n;
	for (i = 0; i < 8; i++)
	{
		OLED_WR_Byte(0xb0 + i, OLED_CMD); // 设置页地址（0~7）
		OLED_WR_Byte(0x00, OLED_CMD);	  // 设置显示位置—列低地址
		OLED_WR_Byte(0x10, OLED_CMD);	  // 设置显示位置—列高地址
		for (n = 0; n < 128; n++)
			OLED_WR_Byte(0, OLED_DATA);
	} // 更新显示
}
// 点亮 OLED 屏幕并填充全屏白色
void OLED_On(void)
{
	u8 i, n;
	for (i = 0; i < 8; i++)
	{
		OLED_WR_Byte(0xb0 + i, OLED_CMD); // 设置页地址（0~7）
		OLED_WR_Byte(0x00, OLED_CMD);	  // 设置显示位置—列低地址
		OLED_WR_Byte(0x10, OLED_CMD);	  // 设置显示位置—列高地址
		for (n = 0; n < 128; n++)
			OLED_WR_Byte(1, OLED_DATA);
	} // 更新显示
}
// 在指定位置显示一个字符,包括部分字符
// x:0~127
// y:0~63
// mode:0,反白显示;1,正常显示
// size:选择字体 16/12
void OLED_ShowChar(u8 x, u8 y, u8 chr)
{
	unsigned char c = 0, i = 0;
	c = chr - ' '; // 得到偏移后的值
	if (x > Max_Column - 1)
	{
		x = 0;
		y = y + 2;
	}
	if (SIZE == 16)
	{
		OLED_Set_Pos(x, y);
		for (i = 0; i < 8; i++)
			OLED_WR_Byte(F8X16[c * 16 + i], OLED_DATA);
		OLED_Set_Pos(x, y + 1);
		for (i = 0; i < 8; i++)
			OLED_WR_Byte(F8X16[c * 16 + i + 8], OLED_DATA);
	}
	else
	{
		OLED_Set_Pos(x, y);
		for (i = 0; i < 6; i++)
			OLED_WR_Byte(F6x8[c][i], OLED_DATA);
	}
}
// m^n函数
u32 oled_pow(u8 m, u8 n)
{
	u32 result = 1;
	while (n--)
		result *= m;
	return result;
}
// 显示2个数字
// x,y :起点坐标
// len :数字的位数
// size:字体大小
// mode:模式	0,填充模式;1,叠加模式
// num:数值(0~4294967295);
void OLED_ShowNum(u8 x, u8 y, u32 num, u8 len, u8 size2)
{
	u8 t, temp;
	u8 enshow = 0;
	for (t = 0; t < len; t++)
	{
		temp = (num / oled_pow(10, len - t - 1)) % 10;
		if (enshow == 0 && t < (len - 1))
		{
			if (temp == 0)
			{
				OLED_ShowChar(x + (size2 / 2) * t, y, ' ');
				continue;
			}
			else
				enshow = 1;
		}
		OLED_ShowChar(x + (size2 / 2) * t, y, temp + '0');
	}
}
// 显示一个字符号串
void OLED_ShowString(u8 x, u8 y, const char *chr)
{
	unsigned char j = 0;
	while (chr[j] != '\0')
	{
		OLED_ShowChar(x, y, chr[j]);
		x += 8;
		if (x > 120)
		{
			x = 0;
			y += 2;
		}
		j++;
	}
	// HAL_Delay(1000);
}
// 显示汉字
void OLED_ShowCHinese(u8 x, u8 y, u8 no)
{
	u8 t, adder = 0;
	OLED_Set_Pos(x, y);
	for (t = 0; t < 16; t++)
	{
		OLED_WR_Byte(Hzk[2 * no][t], OLED_DATA);
		adder += 1;
	}
	OLED_Set_Pos(x, y + 1);
	for (t = 0; t < 16; t++)
	{
		OLED_WR_Byte(Hzk[2 * no + 1][t], OLED_DATA);
		adder += 1;
	}
}
/***********功能描述：显示显示BMP图片128×64起始点坐标(x,y),x的范围0～127，y为页的范围0～7*****************/
void OLED_DrawBMP(unsigned char x0, unsigned char y0, unsigned char x1, unsigned char y1, unsigned char BMP[])
{
	unsigned int j = 0;
	unsigned char x, y;

	if (y1 % 8 == 0)
		y = y1 / 8;
	else
		y = y1 / 8 + 1;
	for (y = y0; y < y1; y++)
	{
		OLED_Set_Pos(x0, y);
		for (x = x0; x < x1; x++)
		{
			OLED_WR_Byte(BMP[j++], OLED_DATA);
		}
	}
}
// 初始化SSD1306
void OLED_Init(void)
{
	HAL_GPIO_WritePin(ST_SDA0_GPIO_Port, ST_SDA0_Pin | ST_SCL0_Pin, GPIO_PIN_SET); // 初始拉高
	HAL_Delay(200);

	// 精简初始化命令
	OLED_WR_Byte(0xAE, OLED_CMD); // 关闭显示
	OLED_WR_Byte(0xD5, OLED_CMD); // 设置时钟分频
	OLED_WR_Byte(0x80, OLED_CMD); // 建议值
	OLED_WR_Byte(0xA8, OLED_CMD); // 多路复用比
	OLED_WR_Byte(0x3F, OLED_CMD); // 1/64
	OLED_WR_Byte(0xD3, OLED_CMD); // 显示偏移
	OLED_WR_Byte(0x00, OLED_CMD);
	OLED_WR_Byte(0x40, OLED_CMD); // 起始行
	OLED_WR_Byte(0x8D, OLED_CMD); // 电荷泵
	OLED_WR_Byte(0x14, OLED_CMD); // 启用电荷泵
	OLED_WR_Byte(0x20, OLED_CMD); // 寻址模式
	OLED_WR_Byte(0x02, OLED_CMD); // 页寻址
	OLED_WR_Byte(0xA1, OLED_CMD); // 段重映射
	OLED_WR_Byte(0xC8, OLED_CMD); // 扫描方向
	OLED_WR_Byte(0xDA, OLED_CMD); // COM引脚
	OLED_WR_Byte(0x12, OLED_CMD);
	OLED_WR_Byte(0x81, OLED_CMD); // 对比度
	OLED_WR_Byte(0xCF, OLED_CMD);
	OLED_WR_Byte(0xD9, OLED_CMD); // 预充电
	OLED_WR_Byte(0xF1, OLED_CMD);
	OLED_WR_Byte(0xDB, OLED_CMD); // VCOMH
	OLED_WR_Byte(0x40, OLED_CMD);
	OLED_WR_Byte(0xA4, OLED_CMD); // 正常显示
	OLED_WR_Byte(0xA6, OLED_CMD); // 非反色
	OLED_WR_Byte(0xAF, OLED_CMD); // 开启显示

	OLED_Clear(); // 清屏
}

// // 延时调整函数（可在运行时调用）
// void Adjust_I2C_Timing(bool increase)
// {
// 	if (increase && i2c_delay_factor < 20)
// 	{
// 		i2c_delay_factor++;
// 	}
// 	else if (!increase && i2c_delay_factor > 1)
// 	{
// 		i2c_delay_factor--;
// 	}
// }

// // 获取当前延时因子（用于调试）
// uint8_t Get_I2C_Delay_Factor(void)
// {
// 	return i2c_delay_factor;
// }